# -*- coding: utf-8 -*-

from django.forms import ModelForm, ValidationError
from django.contrib.formtools.wizard import FormWizard

from fornecedor.models import Fornecedor

class FornecedorPasso2(ModelForm):
    class Meta:
        model = Fornecedor
        exclude = ('estado', 'doc_empresa', 'data_doc_empresa', 'doc_produtos',
                'data_doc_produtos', 'indice', 'estado','usuario')

class FornecedorPasso3(ModelForm):
    class Meta:
        model = Fornecedor
        fields = ('doc_empresa',)

class FornecedorPasso4(ModelForm):
    class Meta:
        model = Fornecedor
        fields = ('doc_produtos',)

class FornecedorHomologar(ModelForm):
    class Meta:
        model = Fornecedor
        fields = ('indice',)

    def __init__(self, *args, **kwargs):
        super(FornecedorHomologar, self).__init__(*args, **kwargs)
        self.fields['indice'].required = True

    def clean_indice(self):
        indice = self.cleaned_data['indice']
        if indice > 1 or indice < 0:
            raise ValidationError('O indíce deve estar entre 0 e 1.')
        return indice


